home *** CD-ROM | disk | FTP | other *** search
- # Source Generated with Decompyle++
- # File: in.pyc (Python 2.4)
-
- '''Unit tests for socket timeout feature.'''
- import unittest
- from test import test_support
- skip_expected = not test_support.is_resource_enabled('network')
- import time
- import socket
-
- class CreationTestCase(unittest.TestCase):
- '''Test case for socket.gettimeout() and socket.settimeout()'''
-
- def setUp(self):
- self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-
-
- def tearDown(self):
- self.sock.close()
-
-
- def testObjectCreation(self):
- self.assertEqual(self.sock.gettimeout(), None, 'timeout not disabled by default')
-
-
- def testFloatReturnValue(self):
- self.sock.settimeout(7.3449999999999998)
- self.assertEqual(self.sock.gettimeout(), 7.3449999999999998)
- self.sock.settimeout(3)
- self.assertEqual(self.sock.gettimeout(), 3)
- self.sock.settimeout(None)
- self.assertEqual(self.sock.gettimeout(), None)
-
-
- def testReturnType(self):
- self.sock.settimeout(1)
- self.assertEqual(type(self.sock.gettimeout()), type(1.0))
- self.sock.settimeout(3.8999999999999999)
- self.assertEqual(type(self.sock.gettimeout()), type(1.0))
-
-
- def testTypeCheck(self):
- self.sock.settimeout(0)
- self.sock.settimeout(0x0L)
- self.sock.settimeout(0.0)
- self.sock.settimeout(None)
- self.assertRaises(TypeError, self.sock.settimeout, '')
- self.assertRaises(TypeError, self.sock.settimeout, u'')
- self.assertRaises(TypeError, self.sock.settimeout, ())
- self.assertRaises(TypeError, self.sock.settimeout, [])
- self.assertRaises(TypeError, self.sock.settimeout, { })
- self.assertRaises(TypeError, self.sock.settimeout, (0.0+0.0j))
-
-
- def testRangeCheck(self):
- self.assertRaises(ValueError, self.sock.settimeout, -1)
- self.assertRaises(ValueError, self.sock.settimeout, -0x1L)
- self.assertRaises(ValueError, self.sock.settimeout, -1.0)
-
-
- def testTimeoutThenBlocking(self):
- self.sock.settimeout(10)
- self.sock.setblocking(1)
- self.assertEqual(self.sock.gettimeout(), None)
- self.sock.setblocking(0)
- self.assertEqual(self.sock.gettimeout(), 0.0)
- self.sock.settimeout(10)
- self.sock.setblocking(0)
- self.assertEqual(self.sock.gettimeout(), 0.0)
- self.sock.setblocking(1)
- self.assertEqual(self.sock.gettimeout(), None)
-
-
- def testBlockingThenTimeout(self):
- self.sock.setblocking(0)
- self.sock.settimeout(1)
- self.assertEqual(self.sock.gettimeout(), 1)
- self.sock.setblocking(1)
- self.sock.settimeout(1)
- self.assertEqual(self.sock.gettimeout(), 1)
-
-
-
- class TimeoutTestCase(unittest.TestCase):
- '''Test case for socket.socket() timeout functions'''
- fuzz = 2.0
-
- def setUp(self):
- self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
- self.addr_remote = ('www.python.org', 80)
- self.addr_local = ('127.0.0.1', 25339)
-
-
- def tearDown(self):
- self.sock.close()
-
-
- def testConnectTimeout(self):
- _timeout = 0.001
- self.sock.settimeout(_timeout)
- _t1 = time.time()
- self.failUnlessRaises(socket.error, self.sock.connect, self.addr_remote)
- _t2 = time.time()
- _delta = abs(_t1 - _t2)
- self.assert_(_delta < _timeout + self.fuzz, 'timeout (%g) is more than %g seconds more than expected (%g)' % (_delta, self.fuzz, _timeout))
-
-
- def testRecvTimeout(self):
- _timeout = 0.02
- self.sock.connect(self.addr_remote)
- self.sock.settimeout(_timeout)
- _t1 = time.time()
- self.failUnlessRaises(socket.error, self.sock.recv, 1024)
- _t2 = time.time()
- _delta = abs(_t1 - _t2)
- self.assert_(_delta < _timeout + self.fuzz, 'timeout (%g) is %g seconds more than expected (%g)' % (_delta, self.fuzz, _timeout))
-
-
- def testAcceptTimeout(self):
- _timeout = 2
- self.sock.settimeout(_timeout)
- self.sock.bind(self.addr_local)
- self.sock.listen(5)
- _t1 = time.time()
- self.failUnlessRaises(socket.error, self.sock.accept)
- _t2 = time.time()
- _delta = abs(_t1 - _t2)
- self.assert_(_delta < _timeout + self.fuzz, 'timeout (%g) is %g seconds more than expected (%g)' % (_delta, self.fuzz, _timeout))
-
-
- def testRecvfromTimeout(self):
- _timeout = 2
- self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
- self.sock.settimeout(_timeout)
- self.sock.bind(self.addr_local)
- _t1 = time.time()
- self.failUnlessRaises(socket.error, self.sock.recvfrom, 8192)
- _t2 = time.time()
- _delta = abs(_t1 - _t2)
- self.assert_(_delta < _timeout + self.fuzz, 'timeout (%g) is %g seconds more than expected (%g)' % (_delta, self.fuzz, _timeout))
-
-
- def testSend(self):
- pass
-
-
- def testSendto(self):
- pass
-
-
- def testSendall(self):
- pass
-
-
-
- def test_main():
- test_support.requires('network')
- test_support.run_unittest(CreationTestCase, TimeoutTestCase)
-
- if __name__ == '__main__':
- test_main()
-
-